package com.njmd.framework.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

/**
 * 
 * @title: MD5Util
 * @description: 加密码工具类
 * 
 * @author: ygw
 * 
 */
public final class MD5Utils {
	private static final String ALGORIGTHM_MD5 = "MD5";

	private static final int CACHE_SIZE = 2048;
	
	/**
	 * <p>
	 * 生成文件MD5值,支持大文件的MD5的生成
	 * <p>
	 * <p>
	 * 在进行文件校验时，文件读取的缓冲大小[CACHE_SIZE]需与该方法的一致，否则校验失败
	 * </p>
	 * 
	 * @param filePath
	 * @return
	 * @throws Exception
	 */
	public static String generateFileMD5(String filePath) throws Exception {
		String md5 = "";
		File file = new File(filePath);
		if (file.exists()) {
			MessageDigest messageDigest = getMD5();
			InputStream in = new FileInputStream(file);
			byte[] cache = new byte[CACHE_SIZE];
			int nRead = 0;
			while ((nRead = in.read(cache)) != -1) {
				messageDigest.update(cache, 0, nRead);
			}
			in.close();
			byte data[] = messageDigest.digest();
			md5 = byteArrayToHexString(data);
		}
		return md5;
	}

	/**
	 * <p>
	 * MD5摘要字节数组转换为16进制字符串
	 * </p>
	 * 
	 * @param data
	 *            MD5摘要
	 * @return
	 */
	private static String byteArrayToHexString(byte[] data) {
		// 用来将字节转换成 16 进制表示的字符
		char hexDigits[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
				'a', 'b', 'c', 'd', 'e', 'f' };
		// 每个字节用 16 进制表示的话，使用两个字符，所以表示成 16 进制需要 32 个字符
		char arr[] = new char[16 * 2];
		int k = 0; // 表示转换结果中对应的字符位置
		// 从第一个字节开始，对 MD5 的每一个字节转换成 16 进制字符的转换
		for (int i = 0; i < 16; i++) {
			byte b = data[i]; // 取第 i 个字节
			// 取字节中高 4 位的数字转换, >>>为逻辑右移，将符号位一起右移
			arr[k++] = hexDigits[b >>> 4 & 0xf];
			// 取字节中低 4 位的数字转换
			arr[k++] = hexDigits[b & 0xf];
		}
		// 换后的结果转换为字符串
		return new String(arr);
	}

	/**
	 * <p>
	 * 获取MD5实例
	 * </p>
	 * 
	 * @return
	 * @throws NoSuchAlgorithmException
	 */
	private static MessageDigest getMD5() throws NoSuchAlgorithmException {
		return MessageDigest.getInstance(ALGORIGTHM_MD5);
	}

	/**
	 * 用于注册与登录
	 * 
	 * @param s
	 *            需要加密的字符串
	 * 
	 * @return 返回加密后的字符串
	 */
	public static String toMD5(String s) {
		char hexDigits[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
				'a', 'b', 'c', 'd', 'e', 'f' };
		try {
			byte[] strTemp = s.getBytes();
			MessageDigest mdTemp = MessageDigest.getInstance("MD5");
			mdTemp.update(strTemp);
			byte[] md = mdTemp.digest();
			int j = md.length;
			char str[] = new char[j * 2];
			int k = 0;
			for (int i = 0; i < j; i++) {
				byte byte0 = md[i];
				str[k++] = hexDigits[byte0 >>> 4 & 0xf];
				str[k++] = hexDigits[byte0 & 0xf];
			}
			return new String(str);
		} catch (Exception e) {
			return null;
		}
	}
	
	public static void main(String[] args) throws Exception{
		System.out.println(generateFileMD5("D:\\apache-tomcat-7.0.42\\webapps\\ROOT\\mdworkstation\\update\\1_64\\MDWorkStation.exe"));
	}
}
